from die import Die
import pygal

# 创建一个6面骰子
die = Die()

# 随机投掷几次骰子，将结果存在一个列表中
results=[]
for roll_num in range(1000):
    result = die.roll()
    results.append(result)

# print(results)

# 分析结果
frequencies = []
for value in range(1,die.num_sides+1):
    frequency = results.count(value)
    frequencies.append(frequency)

print(frequencies)

# 绘制直方图
hist = pygal.Bar()

hist.title = "Results of rolling one D6 1000 times"
# hist.x_labels = ['1','2','3','4','5','6']
hist.x_labels = [x for x in range(1,die.num_sides+1)]
hist.x_title = "Result"
hist.y_title = "Frequency of Result"

hist.add("D6",frequencies)
# 最后生成的图标图片格式必须为svg
hist.render_to_file("die_visual.svg")